Kerry Back
Run GridSearchCV with
Then use OneHotEncoder and make_column_transformer as before.
param_grid = {
"randomforestregressor__max_depth": [3, 4, 5]
}
cv = GridSearchCV(
pipe,
param_grid=param_grid
)
X = data[["roeq", "mom12m", "industry"]]
y = data["rnk"]
cv.fit(X, y)